# Create your views here.

from django.shortcuts import render_to_response
from models import Pedido
from django.contrib.auth.decorators import login_required

@login_required
def index(request):
	dados = {}
	dados["abertas"] = Pedido.objects.filter(data_entrega__isnull=True).count()
	return render_to_response("entrega.html", dados)

from django.shortcuts import get_object_or_404
from django import forms
from models import Entregador

class EntregadorForm(forms.Form):
	entregador = forms.ModelChoiceField(Entregador.objects.all())
	
from django.core.context_processors import csrf
from django.http import HttpResponseRedirect
	

@login_required
def alterar(request, pedido_id):
	pedido = get_object_or_404(Pedido, id=pedido_id)
	dados = {}
	dados.update(csrf(request))
	form = EntregadorForm()
	if request.POST:
		form = EntregadorForm(request.POST)
		if form.is_valid():
			pedido.entregador = form.cleaned_data["entregador"]
			pedido.save()
			return HttpResponseRedirect("/entrega/pedidos/")
	dados["form"] = form
	return render_to_response("alterar.html", dados)

class PedidoForm(forms.ModelForm):
	class Meta:
		model = Pedido

@login_required
def nova(request):
	dados = {}
	dados.update(csrf(request))
	form = PedidoForm()
	if request.POST:
		form = PedidoForm(request.POST)
		if form.is_valid():
			form.save()
			return HttpResponseRedirect("/entrega")
	dados["form"] = form
	return render_to_response("nova.html", dados)

